from django.http import QueryDict
from django.template import Library
from django.conf import settings
from django.urls import reverse
from django.utils.safestring import mark_safe
from web import models

register = Library()

#为客户管理——充值界面 加颜色 num表示用什么颜色样式
@register.filter
def color(num):
    # print("传输编号",num)
    # 找到数据库TransactionRecord 之前定义的颜色映射关系
    return models.TransactionRecord.charge_type_class_mapping[num]

@register.filter
def colors(num):
    # print("传输编号",num)
    # 找到数据库Order 之前定义的颜色映射关系
    return models.Order.charge_type_class_mapping[num]